package com.kalvan.core.db;

import com.alibaba.druid.support.http.StatViewServlet;
import com.alibaba.druid.support.http.WebStatFilter;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.boot.web.servlet.ServletRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.HashMap;
import java.util.Map;

/**
 * @author chenliang
 */
@Configuration
public class DruidConfiguration {

    @Value("${druid.loginUsername:admin}")
    private String loginUsername;
    @Value("${druid.loginPassword:admin}")
    private String loginPassword;
    @Value("${druid.allow:127.0.0.1}")
    private String allow;
    @Value("${druid.deny:}")
    private String deny;
    @Value("${druid.resetEnable:false}")
    private String resetEnable;

    /**
     * 配置一个管理后台的Servlet
     *
     * @return ServletRegistrationBean
     */
    @Bean
    public ServletRegistrationBean statViewServlet() {
        ServletRegistrationBean bean = new ServletRegistrationBean(new StatViewServlet(), "/druid/*");
        Map<String, String> initParams = new HashMap<>();
        initParams.put("loginUsername", loginUsername);
        initParams.put("loginPassword", loginPassword);
        // 默认就是允许所有访问
        initParams.put("allow", allow);
        // 拒绝访问
        initParams.put("deny", deny);
        initParams.put("resetEnable", resetEnable);
        bean.setInitParameters(initParams);
        return bean;
    }

    // 2、配置一个web监控的filter
    @Bean
    public FilterRegistrationBean webStatFilter() {
        FilterRegistrationBean bean = new FilterRegistrationBean(new WebStatFilter());
        bean.addUrlPatterns("/*");
        bean.addInitParameter("exclusions", "*.js,*.gif,*.jpg,*.png,*.css,*.ico,/druid/*");
        return bean;
    }

}
